Mapping data type in Python